package com.runmyai.aitools.model;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;

/**
 * @author wry
 */
@Data
@ToString
@AllArgsConstructor
@NoArgsConstructor
public class Chat {

    private static final String USER = "user";
    private static final String ASSISTANT = "assistant";

    private String role;
    private String content;

    public static Chat user(String content) {
        return new Chat(USER, content);
    }

    public static Chat assistant(String content) {
        return new Chat(ASSISTANT, content);
    }
}